from theseus.base.augmentations import TRANSFORM_REGISTRY